package io.renren.membermanage.bo;

import lombok.Data;

@Data
public class ModelTrainParamBo {

    //用户id
    private Long userId;

    //前端传入模型id，然后去数据库匹配其对应的路径，将路径赋给ModelTrainParamDto中的pretrained_model_name_or_path
    private Long modelId;

    private String output_name;

    private int max_train_epochs;

    private int train_batch_size;

    private int save_every_n_epochs;

    private String sample_prompts;
}
